package com.example.vertification.Utils;


import com.example.vertification.service.EmailService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.FileSystemResource;
import org.springframework.mail.SimpleMailMessage;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.stereotype.Service;
import org.thymeleaf.TemplateEngine;
import org.thymeleaf.context.Context;

import javax.mail.MessagingException;
import javax.mail.internet.MimeMessage;
import java.io.File;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;

/**
 * 发送邮件工具类 MailUtil
 *
 * @author heshi
 * @date 2021/3/22 16:52
 */

@Service
public class MailUtil implements EmailService
{
    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    //Spring Boot 提供了一个发送邮件的简单抽象，使用的是下面这个接口，这里直接注入即可使用
    @Autowired
    private JavaMailSender mailSender;




    @Autowired
    private TemplateEngine templateEngine;



    // 配置文件中我的qq邮箱
    @Value("${spring.mail.from}")
    private String from;

    /**
     * 简单文本邮件
     * @param to 收件人
     * @param subject 主题
     * @param content 内容
     */

    @Override
    public void sendSimpleMail(String to, String subject, String content) {
        //创建SimpleMailMessage对象
        SimpleMailMessage message = new SimpleMailMessage();
        //邮件发送人
        message.setFrom(from);
        //邮件接收人
        message.setTo(to);
        //邮件主题
        message.setSubject(subject);
        //邮件内容
        message.setText(content);
        //发送邮件
        mailSender.send(message);
    }

    /**
     * html邮件
     * @param to 收件人,多个时参数形式 ："xxx@xxx.com,xxx@xxx.com,xxx@xxx.com"
     * @param subject 主题
     * @param content 内容
     */

    @Override
    public void sendHtmlMail(String to, String subject, String content, ArrayList data) {
        // 创建邮件正文模板的上下文
        Context context = new Context();
        context.setVariable("title", subject);
        context.setVariable("content", content);
        context.setVariable("uid", data.get(0));
        context.setVariable("username", data.get(1));
        // 使用Thymeleaf渲染邮件正文
        String htmlContent = templateEngine.process("mail/mail", context);
        // 创建邮件对象
        MimeMessage message = mailSender.createMimeMessage();
        MimeMessageHelper messageHelper;
        try {
            // 设置邮件内容类型为HTML
            messageHelper = new MimeMessageHelper(message, true, StandardCharsets.UTF_8.name());
            // 设置邮件发送人
            messageHelper.setFrom(from);
            // 设置邮件接收人
            messageHelper.setTo(to);
            // 设置邮件主题
            messageHelper.setSubject(subject);
            // 设置邮件内容
            messageHelper.setText(htmlContent, true); // true表示内容是HTML

            // 发送邮件
            mailSender.send(message);

            // 日志信息
            // logger.info("邮件已经发送。");

        } catch (MessagingException e) {
            // logger.error("发送邮件时发生异常！", e);
            throw new RuntimeException("发送邮件时发生异常！", e);
        }
    }




    /**
     * 带附件的邮件
     * @param to 收件人
     * @param subject 主题
     * @param content 内容
     * @param filePath 附件
     */

    @Override
    public void sendAttachmentsMail(String to, String subject, String content, String filePath) {
        MimeMessage message = mailSender.createMimeMessage();
        try {
            MimeMessageHelper helper = new MimeMessageHelper(message, true);
            helper.setFrom(from);
            helper.setTo(to);
            helper.setSubject(subject);
            helper.setText(content, true);

            FileSystemResource file = new FileSystemResource(new File(filePath));
            String fileName = filePath.substring(filePath.lastIndexOf(File.separator));
            helper.addAttachment(fileName, file);
            mailSender.send(message);
            //日志信息
            logger.info("邮件已经发送。");
        } catch (Exception e) {
            logger.error("发送邮件时发生异常！", e);
        }
    }
}